
// 缓冲区类的实现

#ifndef __BUFFER__
#define __BUFFER__

#include <vector>
#include <cassert>
#include "util.hpp"

namespace violet_log
{
    #define DEFAULT_BUFFER_SIZE (10 * 1024 * 1024)//给太大了会导致缓冲区数据很多，如果是滚动大小的落地方式，可能会超出所设定的大小
    #define THRES_BUFFER_SIZE (100 * 1024 * 1024)//阈值
    #define INCREASE_BUFFER_SIZE (10 * 1024 * 1024)
    class buffer
    {
    public:
        buffer():_buffer(DEFAULT_BUFFER_SIZE),_read_idx(0),_write_idx(0){}

        // 向缓冲区中写入数据
        void push(const char *data, size_t len)
        {
            if(len > writeable_size())
            {
                //return;//阻塞
                ensure_enough_size(len);//扩容，用于极限性能测试
            }

            std::copy(data, data + len, &_buffer[_write_idx]);
            move_write(len);
        }

        // 获取可读数据的起始位置
        const char* begin()
        {
            return &_buffer[_read_idx];
        }

        // 获取可读数据的大小
        size_t readable_size()
        {
            return _write_idx - _read_idx;
        }

        // 获取可写数据的大小
        size_t writeable_size()
        {
            return _buffer.size() - _write_idx;
        }

        // 重置读写位置
        void reset()
        {
            _read_idx = 0;
            _write_idx = 0;
        }

        // 移动读位置
        void move_read(size_t len)
        {
            assert(len <= readable_size());
            _read_idx += len;
        }
        
        // 判断缓冲区是否为空
        bool empty()
        {
            return _write_idx == _read_idx;
        }

        // 交换缓冲区
        void swap(buffer& b2)
        {
            _buffer.swap(b2._buffer);
            std::swap(_read_idx, b2._read_idx);
            std::swap(_write_idx, b2._write_idx);
        }

    private:
        //扩容
        void ensure_enough_size(size_t len)
        {
            if(_buffer.size() < THRES_BUFFER_SIZE)
                _buffer.resize(_buffer.size() * 2 + len);
            else
                _buffer.resize(_buffer.size() + INCREASE_BUFFER_SIZE + len);
        }

        // 移动写位置
        void move_write(size_t len)
        {
            assert(len <= writeable_size());
            _write_idx += len;
        }

    private:
        std::vector<char> _buffer; // 缓冲区
        size_t _read_idx;          // 可读数据的起始位置
        size_t _write_idx;         // 可写位置的起始位置
    };
}

#endif